package com.ruoyi.chat.service.impl;

import com.ruoyi.chat.domain.DatasetRole;
import com.ruoyi.chat.mapper.DatasetRoleMapper;
import com.ruoyi.chat.service.IDatasetRoleService;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.CollectionUtils;

import java.util.List;

/**
 * 知识库角色关联Service业务层处理
 * 
 * @author ruoyi
 * @date 2025-03-25
 */
@Service
public class DatasetRoleServiceImpl implements IDatasetRoleService
{
    @Autowired
    private DatasetRoleMapper datasetRoleMapper;

    /**
     * 查询知识库角色关联
     * 
     * @param datasetId 知识库角色关联主键
     * @return 知识库角色关联
     */
    @Override
    public DatasetRole selectDatasetRoleByDatasetId(Long datasetId)
    {
        return datasetRoleMapper.selectDatasetRoleByDatasetId(datasetId);
    }

    /**
     * 查询知识库角色关联列表
     * 
     * @param datasetRole 知识库角色关联
     * @return 知识库角色关联
     */
    @Override
    public List<DatasetRole> selectDatasetRoleList(DatasetRole datasetRole)
    {
        return datasetRoleMapper.selectDatasetRoleList(datasetRole);
    }

    /**
     * 新增知识库角色关联
     * 
     * @param datasetRole 知识库角色关联
     * @return 结果
     */
    @Override
    public int insertDatasetRole(DatasetRole datasetRole)
    {
        return datasetRoleMapper.insertDatasetRole(datasetRole);
    }

    /**
     * 修改知识库角色关联
     * 
     * @param datasetRole 知识库角色关联
     * @return 结果
     */
    @Override
    public int updateDatasetRole(DatasetRole datasetRole)
    {
        return datasetRoleMapper.updateDatasetRole(datasetRole);
    }

    /**
     * 批量删除知识库角色关联
     * 
     * @param datasetIds 需要删除的知识库角色关联主键
     * @return 结果
     */
    @Override
    public int deleteDatasetRoleByDatasetIds(Long[] datasetIds)
    {
        return datasetRoleMapper.deleteDatasetRoleByDatasetIds(datasetIds);
    }

    /**
     * 删除知识库角色关联信息
     * 
     * @param datasetId 知识库角色关联主键
     * @return 结果
     */
    @Override
    public int deleteDatasetRoleByDatasetId(Long datasetId)
    {
        return datasetRoleMapper.deleteDatasetRoleByDatasetId(datasetId);
    }

    @Override
    @Transactional
    public void dataRoleRelate(Long datasetId, List<DatasetRole> roles) {
        // 先清空知识库之前绑定的角色
        datasetRoleMapper.deleteDatasetRoleByDatasetId(datasetId);
        // 重新绑定
        // 如果新绑定的角色为空 则直接返回
        if (CollectionUtils.isEmpty(roles)) {
            return;
        }
        datasetRoleMapper.batchAddDatasetRole(roles);
    }
}
